from selenium.webdriver.common.by import By

from homework.homework_my.po.add_member_page import AddMemberPage
from homework.homework_my.po.base_page import BasePage
from homework.homework_my.po.contact_page import ContactPage


class MainPage(BasePage):
    _base_url = "https://work.weixin.qq.com/wework_admin/frame#index"
    def goto_add_member(self):
        self.driver.find_element(By.CSS_SELECTOR,".ww_indexImg_AddMember").click()
        return AddMemberPage(self.driver)

    def goto_contact(self):
        return ContactPage(self.driver)